export interface Message {
  role: 'user' | 'assistant' | 'system';
  content: string;
  id?: string;
}

export interface ChatRequest {
  message: string;
  system_prompt?: string;
  document_id?: string;
  use_memory?: boolean;
  stream?: boolean;
}

export interface ChatResponse {
  response: string;
  sources?: Array<{
    content: string;
    source: string;
  }>;
}

export interface KnowledgeBase {
  knowledge_base_id: string;
  name: string;
  description: string;
  created_at: string;
  document_count: number;
}

export interface DocumentResponse {
  document_id: string;
  message: string;
  metadata: {
    filename: string;
    content_type: string;
    upload_time: string;
    name: string;
    description: string;
  };
} 